package com.by.controller;

/**
 * @version 1.0
 * @Author 彭彦彬
 * @Date 2025/4/2 21:32
 * @注释
 */


import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

@RestController
public class FileUploadController {

    // 文件存储目录（可以根据需要修改）
    private static final String UPLOAD_DIR = "sc04-uploadPicture/target/classes/static/uploads/";

    /**
     * 文件上传接口
     *
     * @param file 上传的文件
     * @return 响应结果
     */
    @PostMapping("/upload")
    public ResponseEntity<String> uploadFile(@RequestParam("file") MultipartFile file) {
        // 检查文件是否为空
        if (file.isEmpty()) {
            return ResponseEntity.status(HttpStatus.BAD_REQUEST).body("上传失败：文件为空");
        }

        try {
            // 确保上传目录存在
            File uploadDir = new File(UPLOAD_DIR);
            if (!uploadDir.exists()) {
                uploadDir.mkdirs();
            }

            // 获取文件名并保存到指定目录
            String fileName = file.getOriginalFilename();
            Path filePath = Paths.get(UPLOAD_DIR + fileName);
            Files.copy(file.getInputStream(), filePath);

            return ResponseEntity.ok("文件上传成功：" + fileName);
        } catch (IOException e) {
            e.printStackTrace();
            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("上传失败：" + e.getMessage());
        }
    }

    /**
     * 获取所有文件路径接口
     *
     * @return 所有文件路径的列表
     */
    @GetMapping("/api/files/list")
    public ResponseEntity<List<String>> listFiles() {
        // 确保上传目录存在
        File uploadDir = new File(UPLOAD_DIR);
        if (!uploadDir.exists() || !uploadDir.isDirectory()) {
            return ResponseEntity.ok(new ArrayList<>()); // 返回空列表
        }

        // 获取目录中的所有文件
        File[] files = uploadDir.listFiles();
        List<String> filePaths = new ArrayList<>();

        if (files != null) {
            for (File file : files) {
                if (file.isFile()) { // 只处理文件，忽略子目录
                    filePaths.add(file.getAbsolutePath());
                }
            }
        }

        return ResponseEntity.ok(filePaths);
    }

}
